package com.example.arithmeticleetcode.leetcode.dec;

/**
 * @program: arithmetic-leetcode
 * @description:
 * @author: FangZhen
 * @create: 2020-12-09 11:27
 **/
public class PlaceNQueens {

    public static void main(String[] args) {
        new PlaceNQueens().placeNQueens(4);
    }

    /**
     * 数组索引是行号，数组元素是列号
     */
    int[] cols;
    /**
     * 多少种摆法
     */
    int ways;

    void placeNQueens(int n) {
        if (n < 1) return;
        cols = new int[n];
        place(0);
        System.out.println(n + "皇后，一共有" + ways + "种摆法！");
    }

    /**
     * 从第几行开始摆放皇后
     *
     * @param row
     */
    void place(int row) {

        if (row == cols.length) {
            show();
            ways++;
            return;
        }

        for (int col = 0; col < cols.length; col++) {

            if (isValid(row, col)) { //剪枝作用
                //在第row第col列摆放皇后
                cols[row] = col;
                place(row + 1);
            }
        }
    }

    /**
     * 判断第row行第col列是否可以摆放皇后
     * @param row
     * @param col
     * @return
     */
    boolean isValid(int row, int col) {
        for (int i = 0; i < row; i++) {
            //第col列已经存在皇后
            if (cols[i] == col) return false;
            //斜线上的皇后
            if (row - i == Math.abs(col - cols[i])) return false;
        }
        return true;
    }

    void show() {
       for (int row = 0; row < cols.length; row++) {
           for (int col = 0; col < cols.length; col++) {
               if (cols[row] == col) {
                   System.out.print("1 ");
               } else {
                   System.out.print("0 ");
               }
           }
           System.out.println();
       }
        System.out.println("----------------------------------------");
    }
}
